from flask import g

# 自定义一个装饰器，这个装饰器是来验证某些请求是否已经登陆过了。如果已经登陆过继续访问吗（本质上就是一个登录拦截的作用）
def login_required(func):
    def wrapper(*args,**kwargs):
        if g.user_id is not None: # 表示已经登陆过了
            return func(*args,**kwargs)
        else:
            return {'message':'Invalid Token'},401

    return wrapper